import os
import shutil
import subprocess

# 定义文件路径
sources_list_path = '/etc/apt/sources.list'
backup_path = '/etc/apt/sources.list.backup'
template_path = 'template/sources.list'

# 备份原始的sources.list文件
if os.path.exists(sources_list_path):
    shutil.copy(sources_list_path, backup_path)
    print(f"{sources_list_path} 已备份到 {backup_path}")
else:
    print(f"{sources_list_path} 文件不存在，无法备份。")

# 将template/sources.list复制到/etc/apt/目录
if os.path.exists(template_path):
    shutil.copy(template_path, sources_list_path)
    print(f"{template_path} 已复制到 {sources_list_path}")
else:
    print(f"{template_path} 文件不存在，无法复制。")

# 更新软件源
subprocess.run(['sudo', 'apt-get', 'update', '-y'], check=True)
print("软件源已更新。")

# 检查更新后是否需要升级
subprocess.run(['sudo', 'apt-get', 'upgrade', '-y'], check=True)
print("软件已升级。")